from app import app, db, User

# 初始化数据库
with app.app_context():
    print("正在初始化数据库...")
    # 创建所有表
    db.create_all()
    print("数据库表创建完成")
    
    # 检查是否已有管理员用户
    admin_user = User.query.filter_by(username='admin').first()
    if not admin_user:
        print("创建默认管理员用户...")
        admin_user = User(username='admin', role='admin')
        admin_user.set_password('admin123')  # 默认密码
        db.session.add(admin_user)
        db.session.commit()
        print('默认管理员账号: admin')
        print('默认管理员密码: admin123')
    else:
        print("管理员用户已存在")
    
    print("数据库初始化完成！")